import requests

DEESEEK_URL = "http://10.181.8.236:9988/v1/completions"  # 可能需要切换到 /v1/chat/completions
API_KEY = "7183ds"

headers = {
    "Authorization": f"Bearer {API_KEY}",
    "Content-Type": "application/json"
}

data = {
    "model": "/data/huggingface/deepseek-ai/DeepSeek-R1-Distill-Llama-8B",
    "prompt": "你好",
    "max_tokens": 50
}

try:
    response = requests.post(DEESEEK_URL, json=data, headers=headers, timeout=10)
    print("状态码:", response.status_code)
    print("响应:", response.json())
except Exception as e:
    print("请求失败:", e)
